#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <sys/types.h>
#include <wait.h>

using namespace std;
 
const int basesize = 1024;
const int argvnum = 64;
const int envnum = 64;

// 全局的命令行参数表
char *gargv[argvnum];// 全局的环境变量表
char gargc = 0; // 记录环境变量的个数

// 自己维护的环境变量表
char *genv[envnum];

// 

string GetUserName()
{
    string username = getenv("USER");
    return username.empty() ? "None" : username;
}

string GetHostName()
{
    string hostname = getenv("HOSTNAME");
    return hostname.empty() ? "None" : hostname;
}

string GetPWD()
{
    //string pwd = getenv("PWD");
    //return pwd.empty() ? "None" : pwd;
    
    // 系统调用
    // 1. 获取当前工作路径
    if(getcwd(pwd, sizeof(pwd)) == nullptr) return "None";

    // 2. 更新环境变量
    snprintf(pwdenv, sizeof(pwdenv), "PWD=%s", pwd);
    putenv(pwdenv);// PWD=XXXX
    return pwd;
}

string MakeCommandLine()
{
    char command_line[basesize];
    snprintf(command_line, basesize, "[%s@%s %s]# ",\
            GetUserName().c_str(), GetHostName().c_str(), GetPWD().c_str());
    return command_line;

}
void PrintCommandLine()  // 1. 命令行提示符
{
    printf("%s",MakeCommandLine().c_str());
    fflush(stdout);
}

bool GetCommandLine(char command_buffer[], int size)    // 2. 获取用户命令
{
    char *result = fgets(command_buffer, size, stdin);
    if(!result)
    {
        return false;
    }
    command_buffer[strlen(command_buffer)-1] = 0; // 把最后输入的回车字符改为0，即纯净版输入
    if(strlen(command_buffer) == 0) return false;
    return true;
}

void ParseCommandLine(char command_buffer[], int len)  // 3. 分析命令
{
    (void)len;
    memset(gargv, 0, sizeof(gargv)); // 初始化/清空环境变量表里面的内容
    gargc = 0;

    const char *sep = " ";
    
    gargv[gargc++] = strtok(command_buffer, sep); // 放到全局的环境列表里面
    
    // 循环放入环境变量表中
    while((bool)(gargv[gargc++] = strtok(nullptr, sep)));
    gargc--;
}

void debug()
{
    printf("argc: %d\n",gargc);
    for(int i=0; gargv[i]; i++)
    {
        printf("argv[%d]: %s\n", i, gargv[i]);
    }
}

bool ExecuteCommand()    // 4. 执行命令
{
    pid_t id = fork(); // 创建子进程
    if(id < 0) return false;
    if(id == 0)
    {
        // child
        // 1.执行命令 调用系统函数
        //execvp(gargv[0], gargv);
        execvpe(gargv[0], gargv, genv);
        // 2.退出
        exit(1);
    }
    // father
    int status = 0;
    pid_t rid = waitpid(id, &status, 0);
    if(rid > 0)
    {
        return true;
    }
    return false;
}

void AddEnv(const char *item)
{
    int index = 0;
    while(genv[index])
    {
        index++;
    }
    // 指向为空的下标
    genv[index] = (char*)malloc(strlen(item)+1);// 不能使用局部变量，要重新申请
    strncpy(genv[index], item, strlen(item)+1);

    genv[++index] = nullptr;
}

bool CheckAndExecBuiltCommand()
{
    if(strcmp(gargv[0], "cd") == 0)
    {
        if(gargc == 2)
        {
            chdir(gargv[1]);
        }
        return true;
    }
    else if(strcmp(gargv[0], "export") == 0)// 导入环境变量
    {
        if(gargc == 2)
        {
            AddEnv(gargv[1]);
        }
        return true;
    }
    else if(strcmp(gargv[0], "env") == 0)
    {
        for(int i=0; genv[i]; i++)
        {
            printf("%s\n",genv[i]);
        }
        return true;
    }
    return false;
}

void InitEnv()
{
    // 从父shell获取环境变量
    extern char **environ;
    int index = 0;
    while(environ[index])
    {
        genv[index] = (char*)malloc(strlen(environ[index]+1));
        strncpy(genv[index], environ[index], strlen(environ[index])+1);
        index++;
    }
    genv[index] = nullptr;
}

int main()
{
    InitEnv();
    char command_buffer[basesize];
    while(true)
    {
         PrintCommandLine();  // 1. 命令行提示符
         
         if(!GetCommandLine(command_buffer, basesize))    // 2. 获取用户命令
         {
            continue;
         }
         //printf("%s\n",command_buffer);

         ParseCommandLine(command_buffer, strlen(command_buffer));  // 3. 分析命令
         //debug();
        
         if(CheckAndExecBuiltCommand())
         {
             continue;
         }
         
         ExecuteCommand();    // 4. 执行命令
    }
    return 0;
}
